This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

This fixes the following import error for Cirq due to mismatched arguments of DaskFinder.find_spec

# in fresh venv for Python 3.12
$ pip install cirq-core rapids-dask-dependency
$ python -c "import cirq"
...
  File "/tmp/t312/lib/python3.12/site-packages/cirq/_import.py", line 66, in find_spec
    spec = self.finder.find_spec(fullname, path=path, target=target)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: DaskFinder.find_spec() got an unexpected keyword argument 'path'

$ pip uninstall rapids-dask-dependency
$ python -c "import cirq"
# works as expected

xref: googlecolab/colabtools#5141
cc: @raydouglass

Thanks for this fix @pavoljuhas! Do you know what version of RAPIDS is installed on the systems where this issue is being experienced? That changes the branch that we would merge this change into.

Do you know what version of RAPIDS is installed on the systems where this issue is being experienced?

The Colab environment where it shows has rapids_dask_dependency version 24.12.0.
The commit 2304660 which introduced the code is in all branches starting from branch-24.06, so ideally the fix would apply to all branches that are still maintained.

Thank you for quick response!

Thank you for contributing this fix @pavoljuhas !

The only branch under development right now is branch-25.04 - We can definitely merge this into 25.04, but it is very unlikely that we can put out a patch release for 25.02 or earlier. Since that process requires significant time and resources, it's unlikely to happen unless the problem affects a large user base (or is related to security).

We can definitely merge this into 25.04, but it is very unlikely that we can put out a patch release for 25.02 or earlier.

That is fine, I will ask colab maintainers to update the default colab environment once the next patch release of 25.04 is out.
Thank you for quick response!

PS: Is there anything else to do on my part to get this merged?

/ok to test

/merge